Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix transformation for xgboost model #77

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

shanghuiyang
Copy link

This patch fixes the transformation while loading an xgboost model file. When the model file is saved from an xgboost model with param "objective= 'multi:softprob'" or "objective= 'multi:softmax'", the loaded model should use TransformSoftmax rather than TransformRaw. This patch works well on my xgboost model.

@coveralls
Copy link

coveralls commented Dec 2, 2020

Pull Request Test Coverage Report for Build 165

  • 1 of 2 (50.0%) changed or added relevant lines in 1 file are covered.
  • No unchanged relevant lines lost coverage.
  • Overall coverage decreased (-0.05%) to 68.6%

Changes Missing Coverage Covered Lines Changed/Added Lines %
xgensemble_io.go 1 2 50.0%
Totals Coverage Status
Change from base Build 159: -0.05%
Covered Lines: 1798
Relevant Lines: 2621

💛 - Coveralls

@nikolaydubina
Copy link

cc: @dmitryikh

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants